local dmginfo = FindMetaTable("CTakeDamageInfo")

function dmginfo:GetInflictor()
	local wep = self:GetAttacker():GetActiveWeapon()
	return wep
end

function dmginfo:GetWeaponClass()
	local wep = self:GetInflictor()
	local ammo = wep:GetPrimaryAmmoType()
	local class = ""
	
	if ammo == 1 then
		class = "ar"
	elseif ammo == 3 then
		class = "pist"
	elseif ammo == 4 then
		class = "smg"
	elseif ammo == 7 then
		class = "sg"
	end
	
	return class
end
	